from ollama import Client
import time

start = time.time()
client = Client(host='http://localhost:11434')

response = client.chat(model='llama3', messages=[
  {
    'role': 'user',
    'content': '1+1=?',
  }
])
end = time.time()
print(response)
print('cost: ' + str(end - start))